#include <iostream>
#include <vector>
#include <cmath>
#include <cfloat>

#include "util.h"
#include "Hungarian.h"
#include "KalmanTracker.h"

using namespace std;

class Sort
{
public:
    Sort(int max_age, int min_hits, double iou_threshold);
    ~Sort() = default;
    vector<TrackingBox> update(vector<TrackingBox> & dets);

private:
    int max_age_;
    int min_hits_;
    int frame_count_;
    double iou_threshold_;
    vector<KalmanTracker> trackers_;
};